package cn.fansunion.leecode.sum;

import java.util.HashMap;
import java.util.Map;

import cn.fansunion.leecode.kit.Kit;

/**
 * 给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * https://leetcode-cn.com/problems/two-sum/
 * 
 * @author wen.lei@brgroup.com
 *
 *         2021-12-25
 */
public class TwoSum {

    /**
     * @param args
     */
    public static void main(String[] args) {
        final TwoSum ts = new TwoSum();
        int[] twoSum = ts.twoSum(new int[] {7, 1, 5, 3, 6, 9}, 15);
        Kit.print(twoSum);
        int[] twoSum2 = ts.twoSumMap(new int[] {7, 1, 5, 3, 6, 9}, 15);
        Kit.print(twoSum2);

    }

    /**
     * 暴力破解,O(n*n),O(1)
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSum(int[] nums, int target) {
        for (int i = 0; i < nums.length; i++) {
            for (int j = 0; j < nums.length; j++) {
                if (i != j && nums[i] + nums[j] == target) {
                    return new int[] {i, j};
                }
            }
        }
        return null;
    }
    
    /**
     * 暴力破解的优化版-查找表法，遍历的过程记录信息,O(n),O(n)；
     * （有点像"对对碰"，第一次，先放进去；第二次，碰碰，找到了;特性是，可以一边遍历，一边比较）
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSumMap(int[] nums, int target) {
        Map<Integer,Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            //nums[i]自己要匹配的数
            int anotherNum = target-nums[i];
            if(map.containsKey(anotherNum)) {
                return new int[] {i, map.get(anotherNum)};
            }else {
                //没找到，把自己放进去nums[i]
                map.put(nums[i], i);
            }
            
        }
        return null;
    }
    
}
